# -*- coding: UTF-8 -*-
import os
import urllib
import sys

__author__ = 'tzq139'


class FileHelper:
    # 创建新目录
    def __init__(self):
        pass

    def mkdir(self, path):
        path = path.strip()
        # 判断路径是否存在
        # 存在     True
        # 不存在   False
        isExists = os.path.exists(path)
        # 判断结果
        if not isExists:
            os.makedirs(path)
            return True
        else:
            # 如果目录存在则不创建，并提示目录已存在
            print("名为", path, '的文件夹已经创建')
            return False

    # 传入图片地址，文件名，保存单张图片
    def saveImg(self, imageURL, fileName, headers=None):
        try:
            opener = urllib.request.build_opener()
            opener.addheaders =headers
            urllib.request.install_opener(opener)
            urllib.request.urlretrieve(imageURL, fileName)
        except:
            print("Unexpected error:", sys.exc_info()[2])
